import os


def file_size():
    a=0                    #初始化一个变量用来存储文件大小的值
    if os.path.isfile(dir):         #判断是否为文件 isfile的话用getsize命令来获取大小添加到初始变量a当中
        a +=os.path.getsize(dir)
    if os.path.isdir(dir):           #上层判断过后 如果为目录的话使用For循环遍历当前目录，并且初始化给变量dir1
        dir1=os.listdir(dir())
        for i in dir1:
            file=os.path.join(dir,i)    #获取文件i的具体路径 dir是目录 i是目标文件并且用join拼接
            if os.path.isdir(file):
                a+=file_size(file)       #自身递归调用
            if os.path.isfile(file):
                a+=os.path.getsize(file)
    return a
b=input("请输入需要统计大小的目录地址")
file_size(b)
